{
    "cells": [
        {
            "cell_type": "markdown",
            "id": "b9702787-10f3-41b2-bd93-7b40d3e53383",
            "metadata": {},
            "source": [
                "# Resnet Block"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 1,
            "id": "dbd63fbb-f61c-48a4-85a8-1003586be3fc",
            "metadata": {},
            "outputs": [
                {
                    "name": "stderr",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:46.389 | DEBUG    | ttnn:<module>:80 - Loading ttnn configuration overrides from environment variable TTNN_CONFIG_OVERRIDES\n",
                        "2025-08-26 02:15:46.389 | DEBUG    | ttnn:<module>:83 - Initial ttnn.CONFIG:\n",
                        "Config{cache_path=/home/ubuntu/.cache/ttnn,model_cache_path=/home/ubuntu/.cache/ttnn/models,tmp_dir=/tmp/ttnn,enable_model_cache=false,enable_fast_runtime_mode=false,throw_exception_on_fallback=false,enable_logging=false,enable_graph_report=false,enable_detailed_buffer_report=false,enable_detailed_tensor_report=false,enable_comparison_mode=false,comparison_mode_should_raise_exception=false,comparison_mode_pcc=0.9999,root_report_path=generated/ttnn/reports,report_name=std::nullopt,std::nullopt}\n"
                    ]
                }
            ],
            "source": [
                "import os\n",
                "os.environ[\"TTNN_CONFIG_OVERRIDES\"] = \"{\\\"enable_fast_runtime_mode\\\": false}\"\n",
                "import torch\n",
                "import ttnn\n",
                "from ttnn.tracer import trace, visualize\n",
                "from ttnn.model_preprocessing import preprocess_model_parameters, fold_batch_norm2d_into_conv2d"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 2,
            "id": "7bb35807-08e5-4de1-a86f-c960c4ffee81",
            "metadata": {},
            "outputs": [
                {
                    "name": "stderr",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:46.551 | INFO     | ttnn.device:__init__:138 - Using default dispatch core type for this system: DispatchCoreType.WORKER\n"
                    ]
                },
                {
                    "name": "stderr",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:46.551 | INFO     | ttnn.device:__init__:140 - Using default dispatch core axis for this system: DispatchCoreAxis.ROW\n"
                    ]
                },
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:46.506 | info     |          Device | Opening user mode device driver (tt_cluster.cpp:193)\n",
                        "2025-08-26 02:15:46.521 | info     |   SiliconDriver | Harvesting mask for chip 0 is 0x40 (NOC0: 0x40, simulated harvesting mask: 0x0). (cluster.cpp:345)\n",
                        "2025-08-26 02:15:46.538 | info     |   SiliconDriver | Opening local chip ids/pci ids: {0}/[0] and remote chip ids {} (cluster.cpp:204)\n",
                        "2025-08-26 02:15:46.538 | info     |   SiliconDriver | All devices in cluster running firmware version: 18.5.0 (cluster.cpp:185)\n",
                        "2025-08-26 02:15:46.538 | info     |   SiliconDriver | IOMMU: disabled (cluster.cpp:159)\n",
                        "2025-08-26 02:15:46.538 | info     |   SiliconDriver | KMD version: 2.2.0 (cluster.cpp:162)\n",
                        "2025-08-26 02:15:46.538 | info     |   SiliconDriver | Software version 6.0.0, Ethernet FW version 6.15.0 (Device 0) (cluster.cpp:1004)\n",
                        "2025-08-26 02:15:46.539 | info     |   SiliconDriver | Pinning pages for Hugepage: virtual address 0x7d0d00000000 and size 0x40000000 pinned to physical address 0x140000000 (pci_device.cpp:610)\n",
                        "2025-08-26 02:15:46.974 | info     |           Metal | Profiler started on device 0 (device_pool.cpp:204)\n"
                    ]
                }
            ],
            "source": [
                "torch.manual_seed(0)\n",
                "device_params = {\"l1_small_size\": 24576}\n",
                "device = ttnn.CreateDevice(device_id=0, **device_params)"
            ]
        },
        {
            "cell_type": "markdown",
            "id": "162b41bd-cd48-44cc-90cb-0040393ff179",
            "metadata": {},
            "source": [
                "## Torch Module (from torchvision)"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 3,
            "id": "2d48f30d-de73-41a9-83b3-30e7a0dadd89",
            "metadata": {},
            "outputs": [],
            "source": [
                "def conv3x3(in_planes: int, out_planes: int, stride: int = 1, groups: int = 1, dilation: int = 1) -> torch.nn.Conv2d:\n",
                "    \"\"\"3x3 convolution with padding\"\"\"\n",
                "    return torch.nn.Conv2d(\n",
                "        in_planes,\n",
                "        out_planes,\n",
                "        kernel_size=3,\n",
                "        stride=stride,\n",
                "        padding=dilation,\n",
                "        groups=groups,\n",
                "        bias=False,\n",
                "        dilation=dilation,\n",
                "    )\n",
                "\n",
                "\n",
                "class TorchBasicBlock(torch.nn.Module):\n",
                "    expansion: int = 1\n",
                "\n",
                "    def __init__(\n",
                "        self,\n",
                "        inplanes: int,\n",
                "        planes: int,\n",
                "        stride: int = 1,\n",
                "        downsample=None,\n",
                "        groups: int = 1,\n",
                "        base_width: int = 64,\n",
                "        dilation: int = 1,\n",
                "        norm_layer=None,\n",
                "    ) -> None:\n",
                "        super().__init__()\n",
                "        if norm_layer is None:\n",
                "            norm_layer = torch.nn.BatchNorm2d\n",
                "        if groups != 1 or base_width != 64:\n",
                "            raise ValueError(\"BasicBlock only supports groups=1 and base_width=64\")\n",
                "        if dilation > 1:\n",
                "            raise NotImplementedError(\"Dilation > 1 not supported in BasicBlock\")\n",
                "        # Both self.conv1 and self.downsample layers downsample the input when stride != 1\n",
                "        self.conv1 = conv3x3(inplanes, planes, stride)\n",
                "        self.bn1 = norm_layer(planes)\n",
                "        self.relu = torch.nn.ReLU(inplace=True)\n",
                "        self.conv2 = conv3x3(planes, planes)\n",
                "        self.bn2 = norm_layer(planes)\n",
                "        self.downsample = downsample\n",
                "        self.stride = stride\n",
                "\n",
                "    def forward(self, x: torch.Tensor) -> torch.Tensor:\n",
                "        identity = x\n",
                "\n",
                "        out = self.conv1(x)\n",
                "        out = self.bn1(out)\n",
                "        out = self.relu(out)\n",
                "        out = self.conv2(out)\n",
                "        out = self.bn2(out)\n",
                "\n",
                "        if self.downsample is not None:\n",
                "            identity = self.downsample(x)\n",
                "\n",
                "        out += identity\n",
                "        out = self.relu(out)\n",
                "\n",
                "        return out"
            ]
        },
        {
            "cell_type": "markdown",
            "id": "bcdcd7c2-fd79-4e60-ba51-6f2958ce8b04",
            "metadata": {},
            "source": [
                "## Create torch module and preprocess it to get ttnn parameters"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 4,
            "id": "13d5d008-4e85-4e3e-8d80-62397f43b55d",
            "metadata": {},
            "outputs": [
                {
                    "name": "stderr",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:48.497 | DEBUG    | ttnn:manage_config:91 - Set ttnn.CONFIG.enable_logging to False\n",
                        "2025-08-26 02:15:48.497 | DEBUG    | ttnn:manage_config:91 - Set ttnn.CONFIG.enable_comparison_mode to False\n",
                        "2025-08-26 02:15:48.497 | WARNING  | ttnn.model_preprocessing:from_torch:533 - ttnn: model cache can be enabled by passing model_name argument to preprocess_model[_parameters] and setting env variable TTNN_CONFIG_OVERRIDES='{\"enable_model_cache\": true}'\n",
                        "2025-08-26 02:15:48.498 | WARNING  | ttnn.model_preprocessing:_initialize_model_and_preprocess_parameters:483 - Putting the model in eval mode\n",
                        "2025-08-26 02:15:48.499 | DEBUG    | ttnn:manage_config:94 - Restored ttnn.CONFIG.enable_comparison_mode to False\n",
                        "2025-08-26 02:15:48.500 | DEBUG    | ttnn:manage_config:94 - Restored ttnn.CONFIG.enable_logging to False\n"
                    ]
                }
            ],
            "source": [
                "torch_model = TorchBasicBlock(inplanes=64, planes=64, stride=1)\n",
                "input_tensor = ttnn.rand((8, 64, 56, 56), dtype=ttnn.float32, device=device)\n",
                "state_dict = torch_model.state_dict()\n",
                "\n",
                "\n",
                "def create_custom_preprocessor(device):\n",
                "    def custom_preprocessor(torch_model, name, ttnn_module_args):\n",
                "        parameters = {}\n",
                "        conv_weight_1, conv_bias_1 = fold_batch_norm2d_into_conv2d(torch_model.conv1, torch_model.bn1)\n",
                "        parameters[\"conv1\"] = {}\n",
                "        parameters[\"conv2\"] = {}\n",
                "        parameters[\"conv1\"][\"weight\"] = ttnn.from_torch(conv_weight_1, dtype=ttnn.bfloat16)\n",
                "        parameters[\"conv1\"][\"bias\"] = ttnn.from_torch(torch.reshape(conv_bias_1, (1, 1, 1, -1)), dtype=ttnn.bfloat16)\n",
                "        conv_weight_2, conv_bias_2 = fold_batch_norm2d_into_conv2d(torch_model.conv2, torch_model.bn2)\n",
                "        parameters[\"conv2\"][\"weight\"] = ttnn.from_torch(conv_weight_2, dtype=ttnn.bfloat16)\n",
                "        parameters[\"conv2\"][\"bias\"] = ttnn.from_torch(torch.reshape(conv_bias_2, (1, 1, 1, -1)), dtype=ttnn.bfloat16)\n",
                "        return parameters\n",
                "\n",
                "    return custom_preprocessor\n",
                "\n",
                "\n",
                "parameters = preprocess_model_parameters(\n",
                "    initialize_model=lambda: torch_model, custom_preprocessor=create_custom_preprocessor(device), device=None\n",
                ")\n"
            ]
        },
        {
            "cell_type": "markdown",
            "id": "92650dab-d639-4f3b-8239-63b663c5fe0b",
            "metadata": {},
            "source": [
                "## Display the parameters of the module"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 5,
            "id": "3ad0923f-98c9-48b8-bd22-838558c8e211",
            "metadata": {},
            "outputs": [
                {
                    "data": {
                        "text/plain": [
                            "{\n",
                            "  conv1: {\n",
                            "    weight: ttnn.Tensor(shape=Shape([64, 64, 3, 3]), layout=Layout.ROW_MAJOR, dtype=DataType.BFLOAT16),\n",
                            "    bias: ttnn.Tensor(shape=Shape([1, 1, 1, 64]), layout=Layout.ROW_MAJOR, dtype=DataType.BFLOAT16)\n",
                            "  },\n",
                            "  conv2: {\n",
                            "    weight: ttnn.Tensor(shape=Shape([64, 64, 3, 3]), layout=Layout.ROW_MAJOR, dtype=DataType.BFLOAT16),\n",
                            "    bias: ttnn.Tensor(shape=Shape([1, 1, 1, 64]), layout=Layout.ROW_MAJOR, dtype=DataType.BFLOAT16)\n",
                            "  }\n",
                            "}"
                        ]
                    },
                    "execution_count": 5,
                    "metadata": {},
                    "output_type": "execute_result"
                }
            ],
            "source": [
                "parameters"
            ]
        },
        {
            "cell_type": "markdown",
            "id": "eafa0bcd-caa8-4fb7-8d23-73605195256d",
            "metadata": {},
            "source": [
                "## Implement ttnn version of the module. Pass in the parameters into the constructor. "
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 6,
            "id": "2ab662b0-97e5-4645-90db-90d3f00c9f82",
            "metadata": {},
            "outputs": [],
            "source": [
                "class Conv:\n",
                "    def __init__(\n",
                "        self,\n",
                "        conv_params,\n",
                "        input_shape,\n",
                "        parameters,\n",
                "        *,\n",
                "        act_block_h=None,\n",
                "        reshard=False,\n",
                "        deallocate=True,\n",
                "        height_sharding=True,\n",
                "        activation=\"\",\n",
                "        groups=1,\n",
                "        dtype=ttnn.bfloat16,\n",
                "    ) -> None:\n",
                "        self.weights = parameters[\"weight\"]\n",
                "        if \"bias\" in parameters:\n",
                "            self.bias = parameters[\"bias\"]\n",
                "        else:\n",
                "            self.bias = None\n",
                "        self.kernel_size = (self.weights.shape[2], self.weights.shape[3])\n",
                "        self.conv_params = conv_params\n",
                "        self.out_channels = self.weights.shape[0]\n",
                "        self.act_block_h = act_block_h\n",
                "        self.reshard = reshard\n",
                "        self.deallocate = deallocate\n",
                "        self.activation = activation\n",
                "        self.groups = groups\n",
                "        self.dtype = dtype\n",
                "        self.shard_layout = (\n",
                "            ttnn.TensorMemoryLayout.HEIGHT_SHARDED if height_sharding else ttnn.TensorMemoryLayout.BLOCK_SHARDED\n",
                "        )\n",
                "        self.input_shape = input_shape\n",
                "\n",
                "    def __call__(self, device, input_tensor):\n",
                "\n",
                "        conv_config = ttnn.Conv2dConfig(\n",
                "            weights_dtype=ttnn.bfloat16,\n",
                "            activation=self.activation,\n",
                "            shard_layout=self.shard_layout,\n",
                "            deallocate_activation=self.deallocate,\n",
                "        )\n",
                "        if self.act_block_h is not None:\n",
                "            conv_config.act_block_h_override = self.act_block_h\n",
                "\n",
                "        [output_tensor, (_out_height, _out_width), (self.weights, self.bias)] = ttnn.conv2d(\n",
                "            input_tensor=input_tensor,\n",
                "            weight_tensor=self.weights,\n",
                "            bias_tensor=self.bias,\n",
                "            in_channels=self.input_shape[3],\n",
                "            out_channels=self.out_channels,\n",
                "            device=device,\n",
                "            kernel_size=self.kernel_size,\n",
                "            stride=(self.conv_params[0], self.conv_params[1]),\n",
                "            padding=(self.conv_params[2], self.conv_params[3]),\n",
                "            batch_size=self.input_shape[0],\n",
                "            input_height=self.input_shape[1],\n",
                "            input_width=self.input_shape[2],\n",
                "            conv_config=conv_config,\n",
                "            groups=self.groups,\n",
                "            return_output_dim=True,\n",
                "            return_weights_and_bias=True,\n",
                "            dtype=self.dtype,\n",
                "        )\n",
                "\n",
                "        return output_tensor\n",
                "\n",
                "\n",
                "class TTNNBasicBlock:\n",
                "\n",
                "    def __init__(\n",
                "        self,\n",
                "        parameters,\n",
                "    ) -> None:\n",
                "        self.conv1 = Conv([1, 1, 1, 1], [8, 56, 56, 64], parameters=parameters[\"conv1\"])\n",
                "        self.conv2 = Conv([1, 1, 1, 1], [8, 56, 56, 64], parameters=parameters[\"conv2\"])\n",
                "        if \"downsample\" in parameters:\n",
                "            self.downsample = parameters.downsample\n",
                "        else:\n",
                "            self.downsample = None\n",
                "\n",
                "    def __call__(self, x, device):\n",
                "        identity = x\n",
                "\n",
                "        out = self.conv1(device, x)\n",
                "        out = ttnn.relu(out)\n",
                "        out = self.conv2(device, out)\n",
                "\n",
                "        if self.downsample is not None:\n",
                "            identity = self.downsample(x)\n",
                "\n",
                "        identity = ttnn.reshape(identity, out.shape)\n",
                "\n",
                "        identity = ttnn.to_memory_config(\n",
                "            identity,\n",
                "            memory_config=ttnn.get_memory_config(out),\n",
                "            dtype=ttnn.bfloat16,\n",
                "        )\n",
                "        out = ttnn.add(out, identity, memory_config=ttnn.get_memory_config(out))\n",
                "        out = ttnn.relu(out)\n",
                "\n",
                "        return out\n",
                "\n",
                "def run_model(model, input_tensor, device):\n",
                "    input_tensor = ttnn.permute(input_tensor, (0, 2, 3, 1))\n",
                "    input_tensor = ttnn.to_layout(input_tensor, layout=ttnn.TILE_LAYOUT, dtype=ttnn.bfloat16)\n",
                "\n",
                "    output_tensor = model(input_tensor, device)\n",
                "    output_tensor = ttnn.to_memory_config(output_tensor, ttnn.DRAM_MEMORY_CONFIG)\n",
                "    output_tensor = ttnn.permute(output_tensor, (0, 3, 1, 2))\n",
                "    output_tensor = ttnn.reshape(output_tensor, input_tensor.shape)\n",
                "    output_tensor = ttnn.from_device(output_tensor)\n",
                "    output_tensor = ttnn.to_torch(output_tensor)\n",
                "    return output_tensor"
            ]
        },
        {
            "cell_type": "markdown",
            "id": "fb126b07-d9be-4a03-b1f7-4f71c4b6243b",
            "metadata": {},
            "source": [
                "## Run ttnn module and display the traced graph"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 7,
            "id": "42a44ab6",
            "metadata": {},
            "outputs": [
                {
                    "name": "stderr",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:53.785 | DEBUG    | ttnn.tracer:visualize:445 - Dumping graph of the model to None\n"
                    ]
                },
                {
                    "name": "stdout",
                    "output_type": "stream",
                    "text": [
                        "2025-08-26 02:15:48.959 | warning  |              Op | ttnn::to_layout: dtype is specified but the tensor is already in the requested layout! So, the dtype won't be changed! (to_layout_op.cpp:53)\n"
                    ]
                },
                {
                    "data": {
                        "image/svg+xml": [
                            "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
                            "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
                            " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
                            "<!-- Generated by graphviz version 2.43.0 (0)\n",
                            " -->\n",
                            "<!-- Title: %3 Pages: 1 -->\n",
                            "<svg width=\"1247pt\" height=\"2541pt\"\n",
                            " viewBox=\"0.00 0.00 1247.00 2541.00\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
                            "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 2537)\">\n",
                            "<title>%3</title>\n",
                            "<polygon fill=\"white\" stroke=\"transparent\" points=\"-4,4 -4,-2537 1243,-2537 1243,4 -4,4\"/>\n",
                            "<!-- ttnn_input_0 -->\n",
                            "<g id=\"node1\" class=\"node\">\n",
                            "<title>ttnn_input_0</title>\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"614,-2533 459,-2533 459,-2456 614,-2456 614,-2533\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"460.5,-2516.5 460.5,-2531.5 613.5,-2531.5 613.5,-2516.5 460.5,-2516.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"509.5\" y=\"-2521.5\" font-family=\"Linux libertine\" font-size=\"10.00\">ttnn.Tensor</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"460.5,-2456.5 460.5,-2515.5 613.5,-2515.5 613.5,-2456.5 460.5,-2456.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"515.5\" y=\"-2505.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"484\" y=\"-2494.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 64, 56, 56])</text>\n",
                            "<text text-anchor=\"start\" x=\"491\" y=\"-2483.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"507\" y=\"-2472.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"462.5\" y=\"-2461.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</g>\n",
                            "<!-- ttnn.permute_3 -->\n",
                            "<g id=\"node2\" class=\"node\">\n",
                            "<title>ttnn.permute_3</title>\n",
                            "<g id=\"a_node2\"><a xlink:href=\"/operation_buffer_report/6\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"614.5,-2409 378.5,-2409 378.5,-2288 614.5,-2288 614.5,-2409\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"379.5,-2288.5 379.5,-2407.5 459.5,-2407.5 459.5,-2288.5 379.5,-2288.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"381.5\" y=\"-2345.5\" font-family=\"Linux libertine\" font-size=\"10.00\">6: ttnn.permute</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"460.5,-2348.5 460.5,-2407.5 613.5,-2407.5 613.5,-2348.5 460.5,-2348.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"520\" y=\"-2397.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"484\" y=\"-2386.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 64, 56, 56])</text>\n",
                            "<text text-anchor=\"start\" x=\"491\" y=\"-2375.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"507\" y=\"-2364.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"462.5\" y=\"-2353.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"460.5,-2288.5 460.5,-2347.5 613.5,-2347.5 613.5,-2288.5 460.5,-2288.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"515.5\" y=\"-2337.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"484\" y=\"-2326.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"491\" y=\"-2315.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"507\" y=\"-2304.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"462.5\" y=\"-2293.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn_input_0&#45;&gt;ttnn.permute_3 -->\n",
                            "<g id=\"edge1\" class=\"edge\">\n",
                            "<title>ttnn_input_0:#0&#45;&gt;ttnn.permute_3:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M537.5,-2456C537.5,-2438.68 537.5,-2432.02 537.5,-2418.53\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"541,-2418.5 537.5,-2408.5 534,-2418.5 541,-2418.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"553\" y=\"-2430\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.to_layout_6 -->\n",
                            "<g id=\"node3\" class=\"node\">\n",
                            "<title>ttnn.to_layout_6</title>\n",
                            "<g id=\"a_node3\"><a xlink:href=\"/operation_buffer_report/7\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"614,-2241 373,-2241 373,-2120 614,-2120 614,-2241\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"374.5,-2120.5 374.5,-2239.5 459.5,-2239.5 459.5,-2120.5 374.5,-2120.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"376.5\" y=\"-2177.5\" font-family=\"Linux libertine\" font-size=\"10.00\">7: ttnn.to_layout</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"460.5,-2180.5 460.5,-2239.5 613.5,-2239.5 613.5,-2180.5 460.5,-2180.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"520\" y=\"-2229.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"484\" y=\"-2218.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"491\" y=\"-2207.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"507\" y=\"-2196.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"462.5\" y=\"-2185.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"460.5,-2120.5 460.5,-2179.5 613.5,-2179.5 613.5,-2120.5 460.5,-2120.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"515.5\" y=\"-2169.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"484\" y=\"-2158.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"491\" y=\"-2147.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"507\" y=\"-2136.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"462.5\" y=\"-2125.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.permute_3&#45;&gt;ttnn.to_layout_6 -->\n",
                            "<g id=\"edge2\" class=\"edge\">\n",
                            "<title>ttnn.permute_3:#0&#45;&gt;ttnn.to_layout_6:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M537.5,-2288C537.5,-2270.68 537.5,-2264.02 537.5,-2250.53\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"541,-2250.5 537.5,-2240.5 534,-2250.5 541,-2250.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"553\" y=\"-2262\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.conv2d_13 -->\n",
                            "<g id=\"node6\" class=\"node\">\n",
                            "<title>ttnn.conv2d_13</title>\n",
                            "<g id=\"a_node6\"><a xlink:href=\"/operation_buffer_report/8\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"1239,-2073 536,-2073 536,-1919 1239,-1919 1239,-2073\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"537.5,-1920 537.5,-2072 611.5,-2072 611.5,-1920 537.5,-1920\"/>\n",
                            "<text text-anchor=\"start\" x=\"539.5\" y=\"-1993.5\" font-family=\"Linux libertine\" font-size=\"10.00\">8: ttnn.conv2d</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"612.5,-2013 612.5,-2072 930.5,-2072 930.5,-2013 612.5,-2013\"/>\n",
                            "<text text-anchor=\"start\" x=\"754.5\" y=\"-2062\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"718.5\" y=\"-2051\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"725.5\" y=\"-2040\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"741.5\" y=\"-2029\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"697\" y=\"-2018\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"931.5,-2013 931.5,-2072 1084.5,-2072 1084.5,-2013 931.5,-2013\"/>\n",
                            "<text text-anchor=\"start\" x=\"991\" y=\"-2056.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 1</text>\n",
                            "<text text-anchor=\"start\" x=\"958\" y=\"-2045.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([64, 64, 3, 3])</text>\n",
                            "<text text-anchor=\"start\" x=\"958.5\" y=\"-2034.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"957\" y=\"-2023.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1085.5,-2013 1085.5,-2072 1238.5,-2072 1238.5,-2013 1085.5,-2013\"/>\n",
                            "<text text-anchor=\"start\" x=\"1145\" y=\"-2056.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 2</text>\n",
                            "<text text-anchor=\"start\" x=\"1115\" y=\"-2045.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 1, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"1112.5\" y=\"-2034.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"1111\" y=\"-2023.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"612.5,-1920 612.5,-2012 930.5,-2012 930.5,-1920 612.5,-1920\"/>\n",
                            "<text text-anchor=\"start\" x=\"750\" y=\"-2002\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"712.5\" y=\"-1991\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"722\" y=\"-1980\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"741.5\" y=\"-1969\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"694.5\" y=\"-1958\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"614.5\" y=\"-1947\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"716\" y=\"-1936\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"694.5\" y=\"-1925\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"931.5,-1920 931.5,-2012 1084.5,-2012 1084.5,-1920 931.5,-1920\"/>\n",
                            "<text text-anchor=\"start\" x=\"986.5\" y=\"-1985.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 1</text>\n",
                            "<text text-anchor=\"start\" x=\"955\" y=\"-1974.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 576, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"958.5\" y=\"-1963.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"978\" y=\"-1952.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"933.5\" y=\"-1941.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1085.5,-1920 1085.5,-2012 1238.5,-2012 1238.5,-1920 1085.5,-1920\"/>\n",
                            "<text text-anchor=\"start\" x=\"1140.5\" y=\"-1985.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 2</text>\n",
                            "<text text-anchor=\"start\" x=\"1115\" y=\"-1974.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 1, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"1112.5\" y=\"-1963.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"1132\" y=\"-1952.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"1087.5\" y=\"-1941.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.to_layout_6&#45;&gt;ttnn.conv2d_13 -->\n",
                            "<g id=\"edge3\" class=\"edge\">\n",
                            "<title>ttnn.to_layout_6:#0&#45;&gt;ttnn.conv2d_13:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M614.5,-2149.5C649.97,-2149.5 751.77,-2118.01 769.01,-2082.96\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"772.47,-2083.55 771.5,-2073 765.68,-2081.85 772.47,-2083.55\"/>\n",
                            "<text text-anchor=\"middle\" x=\"779\" y=\"-2094\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.reshape_26 -->\n",
                            "<g id=\"node11\" class=\"node\">\n",
                            "<title>ttnn.reshape_26</title>\n",
                            "<g id=\"a_node11\"><a xlink:href=\"/operation_buffer_report/11\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"518.5,-2056.5 278.5,-2056.5 278.5,-1935.5 518.5,-1935.5 518.5,-2056.5\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"279.5,-1936 279.5,-2055 363.5,-2055 363.5,-1936 279.5,-1936\"/>\n",
                            "<text text-anchor=\"start\" x=\"281.5\" y=\"-1993\" font-family=\"Linux libertine\" font-size=\"10.00\">11: ttnn.reshape</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"364.5,-1996 364.5,-2055 517.5,-2055 517.5,-1996 364.5,-1996\"/>\n",
                            "<text text-anchor=\"start\" x=\"424\" y=\"-2045\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"388\" y=\"-2034\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"395\" y=\"-2023\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"411\" y=\"-2012\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"366.5\" y=\"-2001\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"364.5,-1936 364.5,-1995 517.5,-1995 517.5,-1936 364.5,-1936\"/>\n",
                            "<text text-anchor=\"start\" x=\"419.5\" y=\"-1985\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"382\" y=\"-1974\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"395\" y=\"-1963\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"411\" y=\"-1952\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"366.5\" y=\"-1941\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.to_layout_6&#45;&gt;ttnn.reshape_26 -->\n",
                            "<g id=\"edge10\" class=\"edge\">\n",
                            "<title>ttnn.to_layout_6:#0&#45;&gt;ttnn.reshape_26:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M537.5,-2120C537.5,-2072.53 455.22,-2102.05 443.01,-2066.01\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"446.45,-2065.37 441.5,-2056 439.53,-2066.41 446.45,-2065.37\"/>\n",
                            "<text text-anchor=\"middle\" x=\"547\" y=\"-2094\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_7 -->\n",
                            "<g id=\"node4\" class=\"node\">\n",
                            "<title>ttnn_input_7</title>\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"1062.5,-2213.5 954.5,-2213.5 954.5,-2147.5 1062.5,-2147.5 1062.5,-2213.5\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"955.5,-2197.5 955.5,-2212.5 1061.5,-2212.5 1061.5,-2197.5 955.5,-2197.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"981\" y=\"-2202.5\" font-family=\"Linux libertine\" font-size=\"10.00\">ttnn.Tensor</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"955.5,-2148.5 955.5,-2196.5 1061.5,-2196.5 1061.5,-2148.5 955.5,-2148.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"987\" y=\"-2186.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"958.5\" y=\"-2175.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([64, 64, 3, 3])</text>\n",
                            "<text text-anchor=\"start\" x=\"959\" y=\"-2164.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"957.5\" y=\"-2153.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_7&#45;&gt;ttnn.conv2d_13 -->\n",
                            "<g id=\"edge4\" class=\"edge\">\n",
                            "<title>ttnn_input_7:#0&#45;&gt;ttnn.conv2d_13:$1</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M1008.5,-2147.5C1008.5,-2118.01 1008.5,-2108.22 1008.5,-2083.04\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"1012,-2083 1008.5,-2073 1005,-2083 1012,-2083\"/>\n",
                            "<text text-anchor=\"middle\" x=\"1024\" y=\"-2094\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 1</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_8 -->\n",
                            "<g id=\"node5\" class=\"node\">\n",
                            "<title>ttnn_input_8</title>\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"1216.5,-2213.5 1108.5,-2213.5 1108.5,-2147.5 1216.5,-2147.5 1216.5,-2213.5\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1109.5,-2197.5 1109.5,-2212.5 1215.5,-2212.5 1215.5,-2197.5 1109.5,-2197.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"1135\" y=\"-2202.5\" font-family=\"Linux libertine\" font-size=\"10.00\">ttnn.Tensor</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1109.5,-2148.5 1109.5,-2196.5 1215.5,-2196.5 1215.5,-2148.5 1109.5,-2148.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"1141\" y=\"-2186.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"1115.5\" y=\"-2175.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 1, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"1113\" y=\"-2164.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"1111.5\" y=\"-2153.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_8&#45;&gt;ttnn.conv2d_13 -->\n",
                            "<g id=\"edge5\" class=\"edge\">\n",
                            "<title>ttnn_input_8:#0&#45;&gt;ttnn.conv2d_13:$2</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M1162.5,-2147.5C1162.5,-2118.01 1162.5,-2108.22 1162.5,-2083.04\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"1166,-2083 1162.5,-2073 1159,-2083 1166,-2083\"/>\n",
                            "<text text-anchor=\"middle\" x=\"1178\" y=\"-2094\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 2</text>\n",
                            "</g>\n",
                            "<!-- ttnn.relu_16 -->\n",
                            "<g id=\"node7\" class=\"node\">\n",
                            "<title>ttnn.relu_16</title>\n",
                            "<g id=\"a_node7\"><a xlink:href=\"/operation_buffer_report/9\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"929,-1872 550,-1872 550,-1685 929,-1685 929,-1872\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"551.5,-1685.5 551.5,-1870.5 609.5,-1870.5 609.5,-1685.5 551.5,-1685.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"553.5\" y=\"-1775.5\" font-family=\"Linux libertine\" font-size=\"10.00\">9: ttnn.relu</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"610.5,-1778.5 610.5,-1870.5 928.5,-1870.5 928.5,-1778.5 610.5,-1778.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"752.5\" y=\"-1860.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"710.5\" y=\"-1849.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"720\" y=\"-1838.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"739.5\" y=\"-1827.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"692.5\" y=\"-1816.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"612.5\" y=\"-1805.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"714\" y=\"-1794.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"692.5\" y=\"-1783.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"610.5,-1685.5 610.5,-1777.5 928.5,-1777.5 928.5,-1685.5 610.5,-1685.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"748\" y=\"-1767.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"710.5\" y=\"-1756.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"720\" y=\"-1745.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"739.5\" y=\"-1734.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"692.5\" y=\"-1723.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"612.5\" y=\"-1712.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"714\" y=\"-1701.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"692.5\" y=\"-1690.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.conv2d_13&#45;&gt;ttnn.relu_16 -->\n",
                            "<g id=\"edge6\" class=\"edge\">\n",
                            "<title>ttnn.conv2d_13:#0&#45;&gt;ttnn.relu_16:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M771.5,-1919C771.5,-1901.67 770.15,-1895.03 769.67,-1881.54\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"773.17,-1881.44 769.5,-1871.5 766.17,-1881.56 773.17,-1881.44\"/>\n",
                            "<text text-anchor=\"middle\" x=\"787\" y=\"-1893\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.conv2d_23 -->\n",
                            "<g id=\"node10\" class=\"node\">\n",
                            "<title>ttnn.conv2d_23</title>\n",
                            "<g id=\"a_node10\"><a xlink:href=\"/operation_buffer_report/10\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"1236,-1495 527,-1495 527,-1308 1236,-1308 1236,-1495\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"528.5,-1308.5 528.5,-1493.5 608.5,-1493.5 608.5,-1308.5 528.5,-1308.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"530.5\" y=\"-1398.5\" font-family=\"Linux libertine\" font-size=\"10.00\">10: ttnn.conv2d</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"609.5,-1401.5 609.5,-1493.5 927.5,-1493.5 927.5,-1401.5 609.5,-1401.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"751.5\" y=\"-1483.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"709.5\" y=\"-1472.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"719\" y=\"-1461.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"738.5\" y=\"-1450.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"691.5\" y=\"-1439.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"611.5\" y=\"-1428.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"713\" y=\"-1417.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"691.5\" y=\"-1406.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"928.5,-1401.5 928.5,-1493.5 1081.5,-1493.5 1081.5,-1401.5 928.5,-1401.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"988\" y=\"-1461.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 1</text>\n",
                            "<text text-anchor=\"start\" x=\"955\" y=\"-1450.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([64, 64, 3, 3])</text>\n",
                            "<text text-anchor=\"start\" x=\"955.5\" y=\"-1439.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"954\" y=\"-1428.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1082.5,-1401.5 1082.5,-1493.5 1235.5,-1493.5 1235.5,-1401.5 1082.5,-1401.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"1142\" y=\"-1461.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 2</text>\n",
                            "<text text-anchor=\"start\" x=\"1112\" y=\"-1450.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 1, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"1109.5\" y=\"-1439.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"1108\" y=\"-1428.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"609.5,-1308.5 609.5,-1400.5 927.5,-1400.5 927.5,-1308.5 609.5,-1308.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"747\" y=\"-1390.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"709.5\" y=\"-1379.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"719\" y=\"-1368.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"738.5\" y=\"-1357.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"691.5\" y=\"-1346.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"611.5\" y=\"-1335.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"713\" y=\"-1324.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"691.5\" y=\"-1313.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"928.5,-1308.5 928.5,-1400.5 1081.5,-1400.5 1081.5,-1308.5 928.5,-1308.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"983.5\" y=\"-1374\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 1</text>\n",
                            "<text text-anchor=\"start\" x=\"952\" y=\"-1363\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 576, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"955.5\" y=\"-1352\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"975\" y=\"-1341\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"930.5\" y=\"-1330\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1082.5,-1308.5 1082.5,-1400.5 1235.5,-1400.5 1235.5,-1308.5 1082.5,-1308.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"1137.5\" y=\"-1374\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 2</text>\n",
                            "<text text-anchor=\"start\" x=\"1112\" y=\"-1363\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 1, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"1109.5\" y=\"-1352\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"1129\" y=\"-1341\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"1084.5\" y=\"-1330\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.relu_16&#45;&gt;ttnn.conv2d_23 -->\n",
                            "<g id=\"edge7\" class=\"edge\">\n",
                            "<title>ttnn.relu_16:#0&#45;&gt;ttnn.conv2d_23:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M769.5,-1685C769.5,-1603.8 768.58,-1581.01 768.5,-1504.61\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"772,-1504.5 768.5,-1494.5 765,-1504.5 772,-1504.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"785\" y=\"-1587.5\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_17 -->\n",
                            "<g id=\"node8\" class=\"node\">\n",
                            "<title>ttnn_input_17</title>\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"1059.5,-1811.5 951.5,-1811.5 951.5,-1745.5 1059.5,-1745.5 1059.5,-1811.5\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"952.5,-1795.5 952.5,-1810.5 1058.5,-1810.5 1058.5,-1795.5 952.5,-1795.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"978\" y=\"-1800.5\" font-family=\"Linux libertine\" font-size=\"10.00\">ttnn.Tensor</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"952.5,-1746.5 952.5,-1794.5 1058.5,-1794.5 1058.5,-1746.5 952.5,-1746.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"984\" y=\"-1784.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"955.5\" y=\"-1773.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([64, 64, 3, 3])</text>\n",
                            "<text text-anchor=\"start\" x=\"956\" y=\"-1762.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"954.5\" y=\"-1751.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_17&#45;&gt;ttnn.conv2d_23 -->\n",
                            "<g id=\"edge8\" class=\"edge\">\n",
                            "<title>ttnn_input_17:#0&#45;&gt;ttnn.conv2d_23:$1</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M1005.5,-1745.5C1005.5,-1637.43 1005.5,-1607.88 1005.5,-1504.72\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"1009,-1504.5 1005.5,-1494.5 1002,-1504.5 1009,-1504.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"1021\" y=\"-1587.5\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 1</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_18 -->\n",
                            "<g id=\"node9\" class=\"node\">\n",
                            "<title>ttnn_input_18</title>\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"1213.5,-1811.5 1105.5,-1811.5 1105.5,-1745.5 1213.5,-1745.5 1213.5,-1811.5\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1106.5,-1795.5 1106.5,-1810.5 1212.5,-1810.5 1212.5,-1795.5 1106.5,-1795.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"1132\" y=\"-1800.5\" font-family=\"Linux libertine\" font-size=\"10.00\">ttnn.Tensor</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1106.5,-1746.5 1106.5,-1794.5 1212.5,-1794.5 1212.5,-1746.5 1106.5,-1746.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"1138\" y=\"-1784.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"1112.5\" y=\"-1773.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 1, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"1110\" y=\"-1762.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"1108.5\" y=\"-1751.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: ROW_MAJOR</text>\n",
                            "</g>\n",
                            "<!-- ttnn_input_18&#45;&gt;ttnn.conv2d_23 -->\n",
                            "<g id=\"edge9\" class=\"edge\">\n",
                            "<title>ttnn_input_18:#0&#45;&gt;ttnn.conv2d_23:$2</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M1159.5,-1745.5C1159.5,-1637.43 1159.5,-1607.88 1159.5,-1504.72\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"1163,-1504.5 1159.5,-1494.5 1156,-1504.5 1163,-1504.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"1175\" y=\"-1587.5\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 2</text>\n",
                            "</g>\n",
                            "<!-- ttnn.add_32 -->\n",
                            "<g id=\"node13\" class=\"node\">\n",
                            "<title>ttnn.add_32</title>\n",
                            "<g id=\"a_node13\"><a xlink:href=\"/operation_buffer_report/13\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"703,-1261 0,-1261 0,-1074 703,-1074 703,-1261\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"1.5,-1074.5 1.5,-1259.5 64.5,-1259.5 64.5,-1074.5 1.5,-1074.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"3.5\" y=\"-1164.5\" font-family=\"Linux libertine\" font-size=\"10.00\">13: ttnn.add</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"65.5,-1167.5 65.5,-1259.5 383.5,-1259.5 383.5,-1167.5 65.5,-1167.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"207.5\" y=\"-1249.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"165.5\" y=\"-1238.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"175\" y=\"-1227.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"194.5\" y=\"-1216.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"147.5\" y=\"-1205.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"67.5\" y=\"-1194.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"169\" y=\"-1183.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"147.5\" y=\"-1172.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"384.5,-1167.5 384.5,-1259.5 702.5,-1259.5 702.5,-1167.5 384.5,-1167.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"526.5\" y=\"-1249.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 1</text>\n",
                            "<text text-anchor=\"start\" x=\"484.5\" y=\"-1238.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"494\" y=\"-1227.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"513.5\" y=\"-1216.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"466.5\" y=\"-1205.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"386.5\" y=\"-1194.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"488\" y=\"-1183.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"466.5\" y=\"-1172.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"65.5,-1074.5 65.5,-1166.5 702.5,-1166.5 702.5,-1074.5 65.5,-1074.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"362.5\" y=\"-1156.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325\" y=\"-1145.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-1134.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-1123.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"307\" y=\"-1112.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"227\" y=\"-1101.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"328.5\" y=\"-1090.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"307\" y=\"-1079.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.conv2d_23&#45;&gt;ttnn.add_32 -->\n",
                            "<g id=\"edge12\" class=\"edge\">\n",
                            "<title>ttnn.conv2d_23:#0&#45;&gt;ttnn.add_32:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M768.5,-1308C768.5,-1190.32 256.94,-1370.32 225.97,-1270.67\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"229.39,-1269.9 224.5,-1260.5 222.46,-1270.9 229.39,-1269.9\"/>\n",
                            "<text text-anchor=\"middle\" x=\"779\" y=\"-1282\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.to_memory_config_29 -->\n",
                            "<g id=\"node12\" class=\"node\">\n",
                            "<title>ttnn.to_memory_config_29</title>\n",
                            "<g id=\"a_node12\"><a xlink:href=\"/operation_buffer_report/12\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"630,-1667 173,-1667 173,-1513 630,-1513 630,-1667\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"174.5,-1514 174.5,-1666 310.5,-1666 310.5,-1514 174.5,-1514\"/>\n",
                            "<text text-anchor=\"start\" x=\"176.5\" y=\"-1587.5\" font-family=\"Linux libertine\" font-size=\"10.00\">12: ttnn.to_memory_config</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"311.5,-1607 311.5,-1666 629.5,-1666 629.5,-1607 311.5,-1607\"/>\n",
                            "<text text-anchor=\"start\" x=\"453.5\" y=\"-1656\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"411.5\" y=\"-1645\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"424.5\" y=\"-1634\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.FLOAT32</text>\n",
                            "<text text-anchor=\"start\" x=\"440.5\" y=\"-1623\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"396\" y=\"-1612\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"311.5,-1514 311.5,-1606 629.5,-1606 629.5,-1514 311.5,-1514\"/>\n",
                            "<text text-anchor=\"start\" x=\"449\" y=\"-1596\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"411.5\" y=\"-1585\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"421\" y=\"-1574\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"440.5\" y=\"-1563\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"393.5\" y=\"-1552\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"313.5\" y=\"-1541\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"415\" y=\"-1530\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"393.5\" y=\"-1519\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.reshape_26&#45;&gt;ttnn.to_memory_config_29 -->\n",
                            "<g id=\"edge11\" class=\"edge\">\n",
                            "<title>ttnn.reshape_26:#0&#45;&gt;ttnn.to_memory_config_29:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M441.5,-1935C441.5,-1818.7 468.83,-1788.55 470.43,-1677.3\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"473.93,-1677.02 470.5,-1667 466.93,-1676.97 473.93,-1677.02\"/>\n",
                            "<text text-anchor=\"middle\" x=\"459\" y=\"-1893\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.to_memory_config_29&#45;&gt;ttnn.add_32 -->\n",
                            "<g id=\"edge13\" class=\"edge\">\n",
                            "<title>ttnn.to_memory_config_29:#0&#45;&gt;ttnn.add_32:$1</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M470.5,-1513C470.5,-1421.61 444.54,-1389.19 486.5,-1308 499.81,-1282.25 533.75,-1290.55 541.78,-1270.54\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"545.26,-1270.95 543.5,-1260.5 538.36,-1269.77 545.26,-1270.95\"/>\n",
                            "<text text-anchor=\"middle\" x=\"502\" y=\"-1399\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 1</text>\n",
                            "</g>\n",
                            "<!-- ttnn.relu_35 -->\n",
                            "<g id=\"node14\" class=\"node\">\n",
                            "<title>ttnn.relu_35</title>\n",
                            "<g id=\"a_node14\"><a xlink:href=\"/operation_buffer_report/14\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"544,-1027 159,-1027 159,-840 544,-840 544,-1027\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"160.5,-840.5 160.5,-1025.5 224.5,-1025.5 224.5,-840.5 160.5,-840.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"162.5\" y=\"-930.5\" font-family=\"Linux libertine\" font-size=\"10.00\">14: ttnn.relu</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"225.5,-933.5 225.5,-1025.5 543.5,-1025.5 543.5,-933.5 225.5,-933.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"367.5\" y=\"-1015.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325.5\" y=\"-1004.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"335\" y=\"-993.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354.5\" y=\"-982.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"307.5\" y=\"-971.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"227.5\" y=\"-960.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"329\" y=\"-949.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"307.5\" y=\"-938.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"225.5,-840.5 225.5,-932.5 543.5,-932.5 543.5,-840.5 225.5,-840.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"363\" y=\"-922.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325.5\" y=\"-911.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"335\" y=\"-900.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354.5\" y=\"-889.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"307.5\" y=\"-878.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"227.5\" y=\"-867.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"329\" y=\"-856.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"307.5\" y=\"-845.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.add_32&#45;&gt;ttnn.relu_35 -->\n",
                            "<g id=\"edge14\" class=\"edge\">\n",
                            "<title>ttnn.add_32:#0&#45;&gt;ttnn.relu_35:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M384.5,-1074C384.5,-1056.68 384.5,-1050.02 384.5,-1036.53\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"388,-1036.5 384.5,-1026.5 381,-1036.5 388,-1036.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"400\" y=\"-1048\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.to_memory_config_38 -->\n",
                            "<g id=\"node15\" class=\"node\">\n",
                            "<title>ttnn.to_memory_config_38</title>\n",
                            "<g id=\"a_node15\"><a xlink:href=\"/operation_buffer_report/15\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"544,-793 87,-793 87,-639 544,-639 544,-793\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"88.5,-640 88.5,-792 224.5,-792 224.5,-640 88.5,-640\"/>\n",
                            "<text text-anchor=\"start\" x=\"90.5\" y=\"-713.5\" font-family=\"Linux libertine\" font-size=\"10.00\">15: ttnn.to_memory_config</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"225.5,-700 225.5,-792 543.5,-792 543.5,-700 225.5,-700\"/>\n",
                            "<text text-anchor=\"start\" x=\"367.5\" y=\"-782\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325.5\" y=\"-771\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"335\" y=\"-760\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354.5\" y=\"-749\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"307.5\" y=\"-738\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: L1, HEIGHT_SHARDED</text>\n",
                            "<text text-anchor=\"start\" x=\"227.5\" y=\"-727\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Grid: {[(x=0,y=0) &#45; (x=7,y=6)], [(x=0,y=7) &#45; (x=4,y=7)]}</text>\n",
                            "<text text-anchor=\"start\" x=\"329\" y=\"-716\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Shape: [416, 64]</text>\n",
                            "<text text-anchor=\"start\" x=\"307.5\" y=\"-705\" font-family=\"Linux libertine\" font-size=\"10.00\">Shard Orientation: ROW_MAJOR</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"225.5,-640 225.5,-699 543.5,-699 543.5,-640 225.5,-640\"/>\n",
                            "<text text-anchor=\"start\" x=\"363\" y=\"-689\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325.5\" y=\"-678\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"335\" y=\"-667\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354.5\" y=\"-656\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"310\" y=\"-645\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.relu_35&#45;&gt;ttnn.to_memory_config_38 -->\n",
                            "<g id=\"edge15\" class=\"edge\">\n",
                            "<title>ttnn.relu_35:#0&#45;&gt;ttnn.to_memory_config_38:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M384.5,-840C384.5,-823.03 384.5,-816.4 384.5,-803.3\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"388,-803 384.5,-793 381,-803 388,-803\"/>\n",
                            "<text text-anchor=\"middle\" x=\"400\" y=\"-814\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.permute_41 -->\n",
                            "<g id=\"node16\" class=\"node\">\n",
                            "<title>ttnn.permute_41</title>\n",
                            "<g id=\"a_node16\"><a xlink:href=\"/operation_buffer_report/16\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"461.5,-592 219.5,-592 219.5,-471 461.5,-471 461.5,-592\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"220.5,-471.5 220.5,-590.5 306.5,-590.5 306.5,-471.5 220.5,-471.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"222.5\" y=\"-528.5\" font-family=\"Linux libertine\" font-size=\"10.00\">16: ttnn.permute</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"307.5,-531.5 307.5,-590.5 460.5,-590.5 460.5,-531.5 307.5,-531.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"367\" y=\"-580.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325\" y=\"-569.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 1, 25088, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-558.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-547.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"309.5\" y=\"-536.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"307.5,-471.5 307.5,-530.5 460.5,-530.5 460.5,-471.5 307.5,-471.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"362.5\" y=\"-520.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325\" y=\"-509.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 64, 1, 25088])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-498.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-487.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"309.5\" y=\"-476.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.to_memory_config_38&#45;&gt;ttnn.permute_41 -->\n",
                            "<g id=\"edge16\" class=\"edge\">\n",
                            "<title>ttnn.to_memory_config_38:#0&#45;&gt;ttnn.permute_41:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M384.5,-639C384.5,-621.68 384.5,-615.02 384.5,-601.53\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"388,-601.5 384.5,-591.5 381,-601.5 388,-601.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"400\" y=\"-613\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.reshape_44 -->\n",
                            "<g id=\"node17\" class=\"node\">\n",
                            "<title>ttnn.reshape_44</title>\n",
                            "<g id=\"a_node17\"><a xlink:href=\"/operation_buffer_report/17\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"461.5,-424 221.5,-424 221.5,-303 461.5,-303 461.5,-424\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"222.5,-303.5 222.5,-422.5 306.5,-422.5 306.5,-303.5 222.5,-303.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"224.5\" y=\"-360.5\" font-family=\"Linux libertine\" font-size=\"10.00\">17: ttnn.reshape</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"307.5,-363.5 307.5,-422.5 460.5,-422.5 460.5,-363.5 307.5,-363.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"367\" y=\"-412.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"325\" y=\"-401.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([1, 64, 1, 25088])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-390.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-379.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"309.5\" y=\"-368.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"307.5,-303.5 307.5,-362.5 460.5,-362.5 460.5,-303.5 307.5,-303.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"362.5\" y=\"-352.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"331\" y=\"-341.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-330.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-319.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"309.5\" y=\"-308.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.permute_41&#45;&gt;ttnn.reshape_44 -->\n",
                            "<g id=\"edge17\" class=\"edge\">\n",
                            "<title>ttnn.permute_41:#0&#45;&gt;ttnn.reshape_44:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M384.5,-471C384.5,-453.68 384.5,-447.02 384.5,-433.53\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"388,-433.5 384.5,-423.5 381,-433.5 388,-433.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"400\" y=\"-445\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.from_device_47 -->\n",
                            "<g id=\"node18\" class=\"node\">\n",
                            "<title>ttnn.from_device_47</title>\n",
                            "<g id=\"a_node18\"><a xlink:href=\"/operation_buffer_report/18\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"461.5,-256 199.5,-256 199.5,-146 461.5,-146 461.5,-256\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"200.5,-147 200.5,-255 306.5,-255 306.5,-147 200.5,-147\"/>\n",
                            "<text text-anchor=\"start\" x=\"202.5\" y=\"-198.5\" font-family=\"Linux libertine\" font-size=\"10.00\">18: ttnn.from_device</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"307.5,-196 307.5,-255 460.5,-255 460.5,-196 307.5,-196\"/>\n",
                            "<text text-anchor=\"start\" x=\"367\" y=\"-245\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"331\" y=\"-234\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-223\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-212\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<text text-anchor=\"start\" x=\"309.5\" y=\"-201\" font-family=\"Linux libertine\" font-size=\"10.00\">Memory: DRAM, INTERLEAVED</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"307.5,-147 307.5,-195 460.5,-195 460.5,-147 307.5,-147\"/>\n",
                            "<text text-anchor=\"start\" x=\"362.5\" y=\"-185\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"331\" y=\"-174\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"334.5\" y=\"-163\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354\" y=\"-152\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.reshape_44&#45;&gt;ttnn.from_device_47 -->\n",
                            "<g id=\"edge18\" class=\"edge\">\n",
                            "<title>ttnn.reshape_44:#0&#45;&gt;ttnn.from_device_47:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M384.5,-303C384.5,-286.03 384.5,-279.4 384.5,-266.3\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"388,-266 384.5,-256 381,-266 388,-266\"/>\n",
                            "<text text-anchor=\"middle\" x=\"400\" y=\"-277\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "<!-- ttnn.to_torch_49 -->\n",
                            "<g id=\"node19\" class=\"node\">\n",
                            "<title>ttnn.to_torch_49</title>\n",
                            "<g id=\"a_node19\"><a xlink:href=\"/operation_buffer_report/19\" xlink:title=\"&lt;TABLE&gt;\">\n",
                            "<polygon fill=\"#dcdcdc\" stroke=\"black\" points=\"449.5,-99 233.5,-99 233.5,0 449.5,0 449.5,-99\"/>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"234.5,-0.5 234.5,-97.5 319.5,-97.5 319.5,-0.5 234.5,-0.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"236.5\" y=\"-46.5\" font-family=\"Linux libertine\" font-size=\"10.00\">19: ttnn.to_torch</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"320.5,-49.5 320.5,-97.5 448.5,-97.5 448.5,-49.5 320.5,-49.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"367.5\" y=\"-87.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Input 0</text>\n",
                            "<text text-anchor=\"start\" x=\"331.5\" y=\"-76.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Shape([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"335\" y=\"-65.5\" font-family=\"Linux libertine\" font-size=\"10.00\">DataType.BFLOAT16</text>\n",
                            "<text text-anchor=\"start\" x=\"354.5\" y=\"-54.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Layout: TILE</text>\n",
                            "<polygon fill=\"none\" stroke=\"black\" points=\"320.5,-0.5 320.5,-48.5 448.5,-48.5 448.5,-0.5 320.5,-0.5\"/>\n",
                            "<text text-anchor=\"start\" x=\"363\" y=\"-38.5\" font-family=\"Linux libertine\" font-size=\"10.00\">Output 0</text>\n",
                            "<text text-anchor=\"start\" x=\"322.5\" y=\"-27.5\" font-family=\"Linux libertine\" font-size=\"10.00\">torch.Size([8, 56, 56, 64])</text>\n",
                            "<text text-anchor=\"start\" x=\"350.5\" y=\"-16.5\" font-family=\"Linux libertine\" font-size=\"10.00\">torch.bfloat16</text>\n",
                            "<text text-anchor=\"start\" x=\"353.5\" y=\"-5.5\" font-family=\"Linux libertine\" font-size=\"10.00\">torch.strided</text>\n",
                            "</a>\n",
                            "</g>\n",
                            "</g>\n",
                            "<!-- ttnn.from_device_47&#45;&gt;ttnn.to_torch_49 -->\n",
                            "<g id=\"edge19\" class=\"edge\">\n",
                            "<title>ttnn.from_device_47:#0&#45;&gt;ttnn.to_torch_49:$0</title>\n",
                            "<path fill=\"none\" stroke=\"black\" d=\"M384.5,-146C384.5,-128.68 384.5,-122.02 384.5,-108.53\"/>\n",
                            "<polygon fill=\"black\" stroke=\"black\" points=\"388,-108.5 384.5,-98.5 381,-108.5 388,-108.5\"/>\n",
                            "<text text-anchor=\"middle\" x=\"400\" y=\"-120\" font-family=\"Times,serif\" font-size=\"10.00\">0 &#45;&gt; 0</text>\n",
                            "</g>\n",
                            "</g>\n",
                            "</svg>\n"
                        ],
                        "text/plain": [
                            "<graphviz.graphs.Digraph at 0x7d0e2035bd90>"
                        ]
                    },
                    "execution_count": 7,
                    "metadata": {},
                    "output_type": "execute_result"
                }
            ],
            "source": [
                "ttnn_model = TTNNBasicBlock(parameters)\n",
                "with ttnn.tracer.trace():\n",
                "    output_tensor = run_model(ttnn_model, input_tensor, device=device)\n",
                "\n",
                "ttnn.tracer.visualize(output_tensor)"
            ]
        },
        {
            "cell_type": "code",
            "execution_count": 8,
            "id": "8c252ad7-80a4-4d2f-bd47-91b2da1e5cc5",
            "metadata": {},
            "outputs": [],
            "source": [
                "ttnn.close_device(device)"
            ]
        }
    ],
    "metadata": {
        "kernelspec": {
            "display_name": "Python 3 (ipykernel)",
            "language": "python",
            "name": "python3"
        },
        "language_info": {
            "codemirror_mode": {
                "name": "ipython",
                "version": 3
            },
            "file_extension": ".py",
            "mimetype": "text/x-python",
            "name": "python",
            "nbconvert_exporter": "python",
            "pygments_lexer": "ipython3",
            "version": "3.10.12"
        }
    },
    "nbformat": 4,
    "nbformat_minor": 5
}
